package com.justnow.offer;

import java.util.Comparator;
import java.util.List;
import java.util.PriorityQueue;

/**
 * @author justnow
 * Created on 2020-08-20
 * Description
 *
 * 23. 合并K个升序链表
 * 给你一个链表数组，每个链表都已经按升序排列。
 *
 * 请你将所有链表合并到一个升序链表中，返回合并后的链表。
 *
 *
 *
 * 示例 1：
 *
 * 输入：lists = [[1,4,5],[1,3,4],[2,6]]
 * 输出：[1,1,2,3,4,4,5,6]
 * 解释：链表数组如下：
 * [
 *   1->4->5,
 *   1->3->4,
 *   2->6
 * ]
 * 将它们合并到一个有序链表中得到。
 * 1->1->2->3->4->4->5->6
 * 示例 2：
 *
 * 输入：lists = []
 * 输出：[]
 * 示例 3：
 *
 * 输入：lists = [[]]
 * 输出：[]
 *
 *
 * 提示：
 *
 * k == lists.length
 * 0 <= k <= 10^4
 * 0 <= lists[i].length <= 500
 * -10^4 <= lists[i][j] <= 10^4
 * lists[i] 按 升序 排列
 * lists[i].length 的总和不超过 10^4
 */
public class Solution13 {

    /**
     * 方法一，使用小顶堆
     * 时间复杂度：堆元素个数，最能有N = lists.length个，每个堆化的时间复杂度是log(N)，所以时间复杂度是
     * O(NlogN)
     * @param lists
     * @return
     */
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0)
            return null;
        PriorityQueue<ListNode> minHeap = new PriorityQueue<ListNode>(lists.length, new Comparator<ListNode>() {
            @Override
            public int compare(ListNode o1, ListNode o2) {
                if (o1.val < o2.val) return -1;
                else if (o1.val == o2.val) return 0;
                else return 1;

            }
        });

        ListNode dumy = new ListNode(0);
        ListNode p = dumy;
        //首先进入小顶堆的是每个链表的头结点
        for (ListNode list : lists) {
            if (list != null)
                minHeap.add(list);
        }

        while (!minHeap.isEmpty()) {
            //从头结点中找到，最小的节点，然后用p的next指针指向这个节点，在这种操作下，minHeap的最大值
            //只能是链表集合的个数
            p.next = minHeap.poll();
            //p转为下一个节点
            p = p.next;
            //如果此时p的下一个节点不为空，那么就将p的下一节点放入小顶堆中
            if (p.next != null)
                minHeap.add(p.next);
        }
        return dumy.next;
    }



    public static void main(String[] args) {

    }
}

class ListNode {
    int val;
    ListNode next;
    ListNode(int x) {
        val = x;
    }
}